_base_ = ["./voc0712.py", "./schedule_2x.py", "./default_runtime.py"]
# model setting
train_cfg = dict(gaussian_iou=0.7)
test_cfg = None
model = dict(type="CenterNet",
             pretrained="open-mmlab://detectron2/resnet50_caffe",
             backbone=dict(type="ResNet",
                           depth=50,
                           num_stages=4,
                           out_indices=(3, ),
                           frozen_stages=1,
                           norm_cfg=dict(type="BN", requires_grad=False),
                           norm_eval=True,
                           style="caffe"),
             neck=None,
             bbox_head=dict(type="CenterHead",
                            num_classes=20,
                            in_channels=2048,
                            loss_heatmap=dict(type="GaussianFocalLoss",
                                              alpha=2.0,
                                              gamma=4.0,
                                              loss_weight=1),
                            loss_wh_size=dict(type="L1Loss",
                                              reduction="mean",
                                              loss_weight=0.1),
                            loss_ct_off=dict(type="L1Loss",
                                             reduction="mean",
                                             loss_weight=1)))
